package com.itheima.wechat;

import com.google.code.kaptcha.impl.DefaultKaptcha;
import com.itheima.constant.MessageConstant;
import com.itheima.constant.RedisConstant;
import com.itheima.entity.Result;
import com.itheima.utils.JedisUtl;
import com.itheima.utils.SmsUtil;
import com.itheima.utils.ValidateCodeUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.Assert;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import redis.clients.jedis.JedisPool;

import javax.imageio.ImageIO;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.awt.image.BufferedImage;
import java.util.Map;

@RestController
@RequestMapping("/validateCode")
public class ValidateCodeController {

    //spring 容器就得有这个对象
    //<dubbo:annotation package="com.itheima.wechat,com.itheima.utils"></dubbo:annotation>
    @Autowired
    JedisUtl jedisUtl;

    @Autowired
    DefaultKaptcha defaultKaptcha;

    //随机生成一个 下单 验证码  调用阿里 短信工具类发送验证码
    @RequestMapping("/send4Order")
    public Result send4Order(String telephone) {
        //接收到前台发过来的手机号码 随机生成一个验证码xxxx
        // 存到Redis中 key:telephone+"validateCode",value:xxxx
        //用工具类发送验证码到用户手机
        try {
            Integer code = ValidateCodeUtils.generateValidateCode(4);
            //把验证码存到redis中  telephone + RedisConstant.SMS_CODE_KEY_ORDER
            jedisUtl.setex(telephone + RedisConstant.SMS_CODE_KEY_ORDER, 6000, String.valueOf(code));
            //发送验证码


//            SmsUtil.sendSmsCode(telephone, code);


            return Result.OK(MessageConstant.SEND_VALIDATECODE_SUCCESS);
        } catch (Exception e) {
            e.printStackTrace();
            return Result.ERROR(MessageConstant.SEND_VALIDATECODE_FAIL);
        }
    }


    //随机生成一个 登陆 验证码  调用阿里 短信工具类发送验证码  //先判断Redis中的图形验证码是否正确
    @RequestMapping("/send4Login")
    public Result send4Login(@RequestBody Map<String, String> map) {
        //先判断Redis中的图形验证码是否正确
        String telephone = map.get("telephone");
        String deviceId = map.get("deviceId");
        String imgCode = map.get("code");

        String redisImgCode = jedisUtl.get(deviceId);

        if (null == redisImgCode || !redisImgCode.equals(imgCode)) {
            return Result.ERROR("图形验证码输入错误");
        }
        //接收到前台发过来的手机号码 随机生成一个验证码xxxx
        // 存到Redis中 key:telephone+"login",value:xxxx
        //用工具类发送验证码到用户手机
        try {
            Integer code = ValidateCodeUtils.generateValidateCode(4);
            //把验证码存到redis中 telephone + RedisConstant.SMS_CODE_KEY_LOGIN
            jedisUtl.setex(telephone + RedisConstant.SMS_CODE_KEY_LOGIN, 6000, String.valueOf(code));
            //发送验证码


//            SmsUtil.sendSmsCode(telephone, code);


            return Result.OK(MessageConstant.SEND_VALIDATECODE_SUCCESS);
        } catch (Exception e) {
            e.printStackTrace();
            return Result.ERROR(MessageConstant.SEND_VALIDATECODE_FAIL);
        }
    }

    //生成图形验证码
    @RequestMapping("/code/{deviceId}")
    public void createCode(@PathVariable String deviceId, HttpServletResponse response) throws Exception {
        //判断前台传的UUID是否为空
        Assert.notNull(deviceId, "机器码不能为空");
        //每次再次请求页面时，浏览器都会发出新请求，而不是显示缓存的页面。
        response.setHeader("Cache-Control", "no-store, no-cache");
        //设置传输格式
        response.setContentType("image/jpeg");
        //生成文字验证码
        String text = defaultKaptcha.createText();
        //生成图片验证码
        BufferedImage image = defaultKaptcha.createImage(text);
        //生成的验证码写入redis 5分钟有效
        jedisUtl.setex(deviceId, 60 * 5, text);
        //获取输出流
        ServletOutputStream out = response.getOutputStream();
        //将图片写回浏览器
        ImageIO.write(image, "JPEG", out);
    }

}
